How to create an AI text generator in python